package com.lemzdo.community.provider;

import com.aliyun.oss.ClientException;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.model.PutObjectRequest;
import com.lemzdo.community.exception.CustomizeException;
import com.lemzdo.community.exception.ErrCodeEnum;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.InputStream;
import java.util.Date;
import java.util.UUID;

@Component//OSS提供类，提供文件上传所需方法，参数等
public class OSSProvider {

    //引入阿里云OSS所需要的参数
    @Value("${aliyun.oss.endpoint}")
    private String endpoint;
    @Value("${aliyun.oss.yourAccessKeyId}")
    private String accessKeyId;
    @Value("${aliyun.oss.yourAccessKeySecret}")
    private String accessKeySecret;
    @Value("${aliyun.oss.bucketName}")
    private String bucketName;

    /**
     * 上传文件到oss
     * @param inputStream 输入流
     * @param fileName  需要上传的文件名
     * @return  返回url用于显示到前端
     */
    public String upload(InputStream inputStream,String fileName){
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        try {
            //通过随机的uuid来创建自定义的唯一的文件名
            String uuid = UUID.randomUUID().toString().replaceAll("-", "");
            String[] names = fileName.split("[.]");
            String name = uuid + "." + names[names.length - 1];

            // 创建PutObjectRequest对象。
            PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, name, inputStream);

            // 如果需要上传时设置存储类型与访问权限，请参考以下示例代码。
            // ObjectMetadata metadata = new ObjectMetadata();
            // metadata.setHeader(OSSHeaders.OSS_STORAGE_CLASS, StorageClass.Standard.toString());
            // metadata.setObjectAcl(CannedAccessControlList.Private);
            // putObjectRequest.setMetadata(metadata);

            // 上传文件。
            ossClient.putObject(putObjectRequest);
            //设置过期时间 3600L * 1000 * 24 * 365 * 10 表示10年
            Date expiration = new Date(new Date().getTime() + 3600L * 1000 * 24 * 365 * 10);
            //获取访问图片的url
            String url = ossClient.generatePresignedUrl(bucketName, name, expiration).toString();

            //返回url用于显示到前端
            return url;

        } catch (Exception e) {
            e.printStackTrace();
            throw new CustomizeException(ErrCodeEnum.FILE_UPLOAD_FAIL);
        } finally {
            // 关闭OSSClient。
            ossClient.shutdown();
        }

    }




}
